Importing variable from a function of one script into separate script in Python